"""
员工管理接口层: 将员工的增删改查方法, 封装成一个类
"""

import requests

# 定义类
from common.get_token import get_login_token
from config import EMP_ID


class IhrmEmpApi(object):

    # 增加员工
    @staticmethod
    def add_emp(token, json_data):
        resp = requests.post(url="http://ihrm-test.itheima.net/api/sys/user",
                             headers={"Authorization": token},
                             json=json_data)
        return resp

    # 修改员工
    @staticmethod
    def update_emp(emp_id, token, json_data):
        resp = requests.put(url="http://ihrm-test.itheima.net/api/sys/user/" + emp_id,
                            headers={"Authorization": token},
                            json=json_data)
        return resp

    # 查询员工
    @staticmethod
    def search_emp(emp_id, token):
        resp = requests.get(url="http://ihrm-test.itheima.net/api/sys/user/" + emp_id,
                            headers={"Authorization": token})
        return resp

    # 删除员工
    @staticmethod
    def delete_emp(emp_id, token):
        resp = requests.delete(url="http://ihrm-test.itheima.net/api/sys/user/" + emp_id,
                               headers={"Authorization": token})
        return resp


if __name__ == '__main__':
    # 创建token, 只需要创建一次, 后面的接口都可以用. 因此不要封装到接口类里
    token = get_login_token()
    emp_id = "1529714921418326016"

    # 添加员工
    add_data = {"username": "hzb", "mobile": "13732734772", "workNumber": "123412412412"}
    resp = IhrmEmpApi.add_emp(token, add_data)
    print("增加员工", resp.json())

    # 修改员工
    update_data = {"username": "惠主播"}
    resp = IhrmEmpApi.update_emp(emp_id, token, update_data)
    print("修改员工", resp.json())

    # 查询员工
    resp = IhrmEmpApi.search_emp(emp_id, token)
    print("查询员工", resp.json())

    # 删除员工
    resp = IhrmEmpApi.delete_emp(emp_id, token)
    print("删除员工", resp.json())
